"use strict";
/*
 * @Description: MCP 兼容性服务器实现
 *   核心功能：
 *   1. 支持新客户端无状态 HTTP 请求（POST /mcp）
 *   2. 兼容旧客户端 SSE 长连接（GET /sse + POST /messages）
 *   3. 统一处理 JSON-RPC 2.0 协议请求
 *
 * 设计要点：
 * - 新客户端采用无状态设计，每次请求独立处理
 * - 旧客户端保持长连接状态，通过 sessionId 维持会话
 * - 两种模式共享同一套工具注册系统
 *
 * @Author: 莫若省
 * @date: 2025/9/2
 */
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
var __generator = (this && this.__generator) || function (thisArg, body) {
    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
    function verb(n) { return function (v) { return step([n, v]); }; }
    function step(op) {
        if (f) throw new TypeError("Generator is already executing.");
        while (g && (g = 0, op[0] && (_ = 0)), _) try {
            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
            if (y = 0, t) op = [op[0] & 2, t.value];
            switch (op[0]) {
                case 0: case 1: t = op; break;
                case 4: _.label++; return { value: op[1], done: false };
                case 5: _.label++; y = op[1]; op = [0]; continue;
                case 7: op = _.ops.pop(); _.trys.pop(); continue;
                default:
                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
                    if (t[2]) _.ops.pop();
                    _.trys.pop(); continue;
            }
            op = body.call(thisArg, _);
        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
    }
};
Object.defineProperty(exports, "__esModule", { value: true });
var express_1 = require("express");
var cors_1 = require("cors");
// 导入 MCP 协议传输层实现
var streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
var sse_js_1 = require("@modelcontextprotocol/sdk/server/sse.js");
// 导入 MCP 服务生成器
var index_js_1 = require("./lib/mcp/index.js");
// 初始化基础 MCP 服务实例（包含已注册的工具）
var server = (0, index_js_1.generateMcpServer)();
// ======================
// Express 应用配置
// ======================
var app = (0, express_1.default)();
// 中间件配置
app.use(express_1.default.json()); // 启用 JSON 请求体解析
app.use((0, cors_1.default)()); // 启用跨域资源共享
// ======================
// 传统 SSE 模式实现 (旧客户端兼容)
// ======================
/**
 * SSE 会话存储表
 * 用于维护长连接客户端的会话状态
 *
 * 数据结构：
 * {
 *   [sessionId: string]: SSEServerTransport 实例
 * }
 */
var sseTransports = {};
/**
 * SSE 连接建立端点 (GET /sse)
 *
 * 工作流程：
 * 1. 创建 SSE 传输层实例
 * 2. 注册会话到全局管理表
 * 3. 绑定连接关闭事件处理
 * 4. 连接 MCP 服务核心
 */
app.get("/sse", function (req, res) { return __awaiter(void 0, void 0, void 0, function () {
    var transport;
    return __generator(this, function (_a) {
        switch (_a.label) {
            case 0:
                transport = new sse_js_1.SSEServerTransport("/messages", res);
                // 会话状态管理
                sseTransports[transport.sessionId] = transport; // 注册会话
                res.on("close", function () {
                    delete sseTransports[transport.sessionId]; // 清理会话资源
                });
                // 连接 MCP 核心服务
                return [4 /*yield*/, server.connect(transport)];
            case 1:
                // 连接 MCP 核心服务
                _a.sent();
                return [2 /*return*/];
        }
    });
}); });
/**
 * SSE 消息接收端点 (POST /messages)
 *
 * 工作流程：
 * 1. 从查询参数获取 sessionId
 * 2. 查找对应的传输层实例
 * 3. 处理消息并返回响应
 */
app.post("/messages", function (req, res) { return __awaiter(void 0, void 0, void 0, function () {
    var sessionId, transport;
    return __generator(this, function (_a) {
        switch (_a.label) {
            case 0:
                sessionId = req.query.sessionId;
                transport = sseTransports[sessionId];
                if (!transport) {
                    return [2 /*return*/, res.status(400).json({ error: "Invalid or expired sessionId" })];
                }
                return [4 /*yield*/, transport.handlePostMessage(req, res, req.body)];
            case 1:
                _a.sent();
                return [2 /*return*/];
        }
    });
}); });
// ======================
// 无状态 HTTP 模式 (新客户端规范)
// ======================
/**
 * 标准 MCP 接入端点 (POST /mcp)
 *
 * 设计特点：
 * - 无会话状态（stateless）
 * - 每次请求独立创建服务实例
 * - 符合 HTTP/REST 最佳实践
 *
 * 工作流程：
 * 1. 创建无状态传输层实例
 * 2. 绑定连接关闭处理
 * 3. 初始化新服务实例
 * 4. 处理 JSON-RPC 请求
 */
app.all("/mcp", function (req, res) { return __awaiter(void 0, void 0, void 0, function () {
    var transport_1, statelessServer, err_1;
    return __generator(this, function (_a) {
        switch (_a.label) {
            case 0:
                _a.trys.push([0, 3, , 4]);
                transport_1 = new streamableHttp_js_1.StreamableHTTPServerTransport({
                    sessionIdGenerator: undefined, // 关键配置：禁用会话状态
                });
                // 请求生命周期管理
                res.on("close", function () { return transport_1.close(); });
                statelessServer = (0, index_js_1.generateMcpServer)();
                return [4 /*yield*/, statelessServer.connect(transport_1)];
            case 1:
                _a.sent();
                // 处理 JSON-RPC 协议请求
                return [4 /*yield*/, transport_1.handleRequest(req, res, req.body)];
            case 2:
                // 处理 JSON-RPC 协议请求
                _a.sent();
                return [3 /*break*/, 4];
            case 3:
                err_1 = _a.sent();
                console.error("[MCP] 请求处理异常:", err_1);
                // 标准化错误响应（遵循 JSON-RPC 2.0 规范）
                if (!res.headersSent) {
                    res.status(500).json({
                        jsonrpc: "2.0",
                        error: {
                            code: -32603, // 内部错误代码
                            message: "Internal server error",
                            data: process.env.NODE_ENV === 'development' ? JSON.stringify(err_1) : undefined
                        },
                        id: null, // 表示通知类错误
                    });
                }
                return [3 /*break*/, 4];
            case 4: return [2 /*return*/];
        }
    });
}); });
// ======================
// 服务启动与监控
// ======================
var PORT = process.env.PORT ? Number(process.env.PORT) : 8848;
app.listen(PORT, function () {
    console.log("[MCP] \u517C\u5BB9\u670D\u52A1\u5DF2\u542F\u52A8\uFF1A\n  \u65B0\u5BA2\u6237\u7AEF\u7AEF\u70B9\uFF1APOST http://localhost:".concat(PORT, "/mcp\n  \u65E7\u5BA2\u6237\u7AEF\u7AEF\u70B9\uFF1AGET  http://localhost:").concat(PORT, "/sse\n  \u76D1\u63A7\u7AEF\u70B9\uFF1AGET  http://localhost:").concat(PORT, "/healthz"));
});
// 简易健康检查端点
app.get('/healthz', function (req, res) {
    res.status(200).json({
        status: 'healthy',
        timestamp: new Date().toISOString()
    });
});
